package xiaomogu.task_engine.core.context;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.slf4j.helpers.FormattingTuple;
import org.slf4j.helpers.MessageFormatter;
import xiaomogu.task_engine.core.log.JobFileAppender;
import xiaomogu.task_engine.core.util.DateUtil;

import java.io.PrintWriter;
import java.io.StringWriter;
import java.util.Date;

/**
 * helper for xxl-job
 *
 * @author xuxueli 2020-11-05
 */
public class JobHelper {

    // ---------------------- base info ----------------------

    private static Logger logger = LoggerFactory.getLogger("xxl-job logger");

    /**
     * current JobId
     *
     * @return
     */
    public static long getJobId(JobContext jobContext) {

        if (jobContext == null) {
            return -1;
        }

        return jobContext.getJobId();
    }

    // ---------------------- for log ----------------------

    /**
     * current JobParam
     *
     * @return
     */
    public static String getJobParam(JobContext jobContext) {

        if (jobContext == null) {
            return null;
        }

        return jobContext.getJobParam();
    }

    // ---------------------- for shard ----------------------

    /**
     * current JobLogFileName
     *
     * @return
     */
    public static String getJobLogFileName(JobContext jobContext) {

        if (jobContext == null) {
            return null;
        }

        return jobContext.getJobLogFileName();
    }

    /**
     * current ShardIndex
     *
     * @return
     */
    public static int getShardIndex(JobContext jobContext) {

        if (jobContext == null) {
            return -1;
        }

        return jobContext.getShardIndex();
    }

    // ---------------------- tool for log ----------------------

    /**
     * current ShardTotal
     *
     * @return
     */
    public static int getShardTotal(JobContext jobContext) {

        if (jobContext == null) {
            return -1;
        }

        return jobContext.getShardTotal();
    }

    /**
     * append log with pattern
     *
     * @param appendLogPattern   like "aaa {} bbb {} ccc"
     * @param appendLogArguments like "111, true"
     */
    public static boolean log(JobContext jobContext, String appendLogPattern, Object... appendLogArguments) {

        FormattingTuple formattingTuple = MessageFormatter.arrayFormat(appendLogPattern, appendLogArguments);
        String appendLog = formattingTuple.getMessage();


        StackTraceElement callInfo = new Throwable().getStackTrace()[1];
        return logDetail(jobContext, callInfo, appendLog);
    }


    public static boolean log(JobContext jobContext, Throwable e) {

        StringWriter stringWriter = new StringWriter();
        e.printStackTrace(new PrintWriter(stringWriter));
        String appendLog = stringWriter.toString();

        StackTraceElement callInfo = new Throwable().getStackTrace()[1];
        return logDetail(jobContext, callInfo, appendLog);
    }


    private static boolean logDetail(JobContext jobContext, StackTraceElement callInfo, String appendLog) {

        if (jobContext == null) {
            return false;
        }


        StringBuffer stringBuffer = new StringBuffer();
        stringBuffer.append(DateUtil.formatDateTime(new Date())).append(" ")
                .append("[" + callInfo.getClassName() + "#" + callInfo.getMethodName() + "]").append("-")
                .append("[" + callInfo.getLineNumber() + "]").append("-")
                .append("[" + Thread.currentThread().getName() + "]").append(" ")
                .append(appendLog != null ? appendLog : "");
        String formatAppendLog = stringBuffer.toString();

        // appendlog
        String logFileName = jobContext.getJobLogFileName();

        if (logFileName != null && logFileName.trim().length() > 0) {
            JobFileAppender.appendLog(logFileName, formatAppendLog);
            return true;
        } else {
            logger.info(">>>>>>>>>>> {}", formatAppendLog);
            return false;
        }
    }

    // ---------------------- tool for handleResult ----------------------

    /**
     * handle success
     *
     * @return
     */
    public static boolean handleSuccess(JobContext jobContext) {
        return handleResult(jobContext, JobContext.HANDLE_COCE_SUCCESS, null);
    }

    /**
     * handle success with log msg
     *
     * @param handleMsg
     * @return
     */
    public static boolean handleSuccess(JobContext jobContext, String handleMsg) {
        return handleResult(jobContext, JobContext.HANDLE_COCE_SUCCESS, handleMsg);
    }

    /**
     * handle fail
     *
     * @return
     */
    public static boolean handleFail(JobContext jobContext) {
        return handleResult(jobContext, JobContext.HANDLE_COCE_FAIL, null);
    }

    /**
     * handle fail with log msg
     *
     * @param handleMsg
     * @return
     */
    public static boolean handleFail(JobContext jobContext, String handleMsg) {
        return handleResult(jobContext, JobContext.HANDLE_COCE_FAIL, handleMsg);
    }

    /**
     * handle timeout
     *
     * @return
     */
    public static boolean handleTimeout(JobContext jobContext) {
        return handleResult(jobContext, JobContext.HANDLE_COCE_TIMEOUT, null);
    }

    /**
     * handle timeout with log msg
     *
     * @param handleMsg
     * @return
     */
    public static boolean handleTimeout(JobContext jobContext, String handleMsg) {
        return handleResult(jobContext, JobContext.HANDLE_COCE_TIMEOUT, handleMsg);
    }

    /**
     * @param handleCode 200 : success
     *                   500 : fail
     *                   502 : timeout
     * @param handleMsg
     * @return
     */
    public static boolean handleResult(JobContext jobContext, int handleCode, String handleMsg) {

        if (jobContext == null) {
            return false;
        }

        jobContext.setHandleCode(handleCode);
        if (handleMsg != null) {
            jobContext.setHandleMsg(handleMsg);
        }
        return true;
    }


}
